import java.util.concurrent.ConcurrentHashMap;

public class UserRepository {
    private static final ConcurrentHashMap<String, User> userDB = new ConcurrentHashMap<>();

    // 原查询方法保持不变
    public User findByUsername(String username) {
        return userDB.get(username);
    }

    public boolean usernameExists(String username) {
        return userDB.containsKey(username);
    }

    // 原注册方法保持不变
    public void register(User user) {
        userDB.put(user.getUsername(), user);
    }

    // 新增日记保存方法（已存在）
    public void saveDiary(String username, Diary diary) {
        User user = userDB.get(username);
        if (user != null) user.addDiary(diary);
    }

    // 新增删除日记方法（关键修正）
    public void deleteDiary(String username, String diaryId) {
        User user = userDB.get(username);
        if (user != null) {
            user.removeDiary(diaryId); // 调用User类的新方法
        }
    }
}